
#include <iostream>
#include <string>
#include "Stock.hh"


/**
* 使用struct定义一种类型Stock，Stock类型至少有两个成员：股票代码、市值，类型定义完成后做以下工作：
* 1. 定义一个数组，容纳一定数量的Stock;
* 2. 初始化每个数组元素
* 3. 一个操作界面，提供一些操作：
*     A. 列出所有股票的信息
*     B. 提示用户选择要更新的股票
*     C. 提示用户输入股票更新内容（如更新市值等）
*     D. 更新成功后打印更新后的股票信息
*     E. [可选功能] 重复A-D操作，直至用户输入"quit"退出程序
*/
using namespace std;

static void show_op(void)
{
    cout << "Stock OP:" << endl;
    cout << "A 列出所有股票的信息\n";
    cout << "B 选择要更新的股票\n";
    cout << "C 输入股票更新市值\n";
    cout << "D 更新后的股票信息\n";
}

int main(void)
{

    int code = 0;
    float value = 0.0;

    while (1)
    {
        show_op();

        string str;
        cin >> str;

        if (!str.compare("quit"))
        {
            cout << "Quit Stock!\n";
            return 0;
        }
        else if (!str.compare("A"))
        {
            StockList();
        }
        else if (!str.compare("B"))
        {
            cout << "选择要更新的股票:\n";
            cin >> code;
        }
        else if (!str.compare("C"))
        {
            cout << "输入股票更新市值\n";
            cin >> value;

            StockUpdate(code, value);
        }
        else if (!str.compare("D"))
        {
            StockList();
        }
        else
        {
            cout << "error\n";
        }
    }

    return 0;
}